from marshmallow import fields
from gis.common.django_ext.forms import BaseSchema, PaginationSchema


class GetBlackListSchema(PaginationSchema):
    mch_id = fields.Integer()
    user_id = fields.String()
    agency_id = fields.Integer()
    agency_name = fields.String()
    created_at_begin = fields.DateTime()
    created_at_end = fields.DateTime()


class AddBlackListSchema(BaseSchema):
    user_id = fields.String(required=True)
    mch_id = fields.Integer(required=True)
    remark = fields.String()
